python solve equations